from pydantic import BaseModel, Field
from pydantic_settings import BaseSettings, SettingsConfigDict


class EmbeddingModel(BaseModel):
    model: str = 'text-embedding-3-small'
    keys: list[str] = Field(default_factory=list)


class LLM(BaseModel):
    embeddings: dict[str, EmbeddingModel] = Field(default_factory=dict)


class LLMSettings(BaseSettings):
    llm: LLM = Field(default_factory=lambda: LLM())

    model_config = SettingsConfigDict(env_prefix='my_prefix_', env_nested_delimiter='__')


# export my_prefix_llm__embeddings__openai__keys='["sk-..."]'
# export my_prefix_llm__embeddings__qwen__keys='["sk-..."]'
print(LLMSettings().model_dump()) # raise error